package _interview150;

import java.util.*;

/**
 * 133. 克隆图
 */
public class No133 {
    private Map<Node, Node> map;

    /**
     * 1. 深度优先搜索
     */
    public Node cloneGraph1(Node node) {
        if (node == null) return null;
        map = new HashMap<>();
        return dfs(node);
    }

    private Node dfs(Node node) {
        // 如果该节点已经被访问过了，则直接从哈希表中取出对应的克隆节点返回
        if (map.containsKey(node)) return map.get(node);
        // 克隆节点，注意到为了深拷贝我们不会克隆它的邻居的列表
        Node cloneNode = new Node(node.val, new ArrayList<>());
        // 哈希表存储
        map.put(node, cloneNode);
        // 遍历该节点的邻居并更新克隆节点的邻居列表
        for (Node neighbor : node.neighbors) {
            cloneNode.neighbors.add(dfs(neighbor));
        }
        return cloneNode;
    }

    /**
     * 2. 广度优先遍历
     */
    public Node cloneGraph2(Node node) {
        if (node == null) return null;

        map = new HashMap<>();
        // 克隆第一个节点并存储到哈希表中
        map.put(node, new Node(node.val, new ArrayList<>()));

        Queue<Node> queue = new LinkedList<>();
        // 将题目给定的节点添加到队列
        queue.offer(node);

        while (!queue.isEmpty()) {
            // 取出队列的头节点
            Node poll = queue.poll();
            // 遍历该节点的邻居
            for (Node neighbor : poll.neighbors) {
                if (!map.containsKey(neighbor)) {
                    // 如果没有被访问过，就克隆并存储在哈希表中
                    map.put(neighbor, new Node(neighbor.val, new ArrayList<>()));
                    // 将邻居节点加入队列中
                    queue.add(neighbor);
                }
                // 更新当前节点的邻居列表
                map.get(poll).neighbors.add(map.get(neighbor));
            }
        }
        return map.get(node);
    }

    class Node {
        public int val;
        public List<Node> neighbors;

        public Node() {
            val = 0;
            neighbors = new ArrayList<>();
        }

        public Node(int _val) {
            val = _val;
            neighbors = new ArrayList<>();
        }

        public Node(int _val, ArrayList<Node> _neighbors) {
            val = _val;
            neighbors = _neighbors;
        }
    }
}
